package com.prj.ufdm.cloud.ms.core.util;

import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.ss.usermodel.RichTextString;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.HSSFColor;

public class CoreExcelExportUtil {

	// 导出的excel标题
	private String title;
	// 导出的excel工作表名
	private String sheetName;
	// 导出的excel列名
	private String[] rowName;
	// 导出的excel数据
	private List<Object[]> dataList = new ArrayList<Object[]>();

	// 不使用标题的构造
	public CoreExcelExportUtil(String sheetName, String[] rowName, List<Object[]> dataList) {
		this.sheetName = sheetName;
		this.rowName = rowName;
		this.dataList = dataList;
	}

	// 使用标题的构造
	public CoreExcelExportUtil(String title, String sheetName, String[] rowName, List<Object[]> dataList) {
		this.title = title;
		this.sheetName = sheetName;
		this.rowName = rowName;
		this.dataList = dataList;
	}

	/*
	 * 导出数据
	 */
	public void export(OutputStream out) throws IOException {

		HSSFWorkbook workbook = new HSSFWorkbook(); // 创建工作簿对象
		HSSFSheet sheet = workbook.createSheet(sheetName); // 创建工作表

		// sheet样式定义【getColumnTopStyle()/getStyle()均为自定义方法 - 在下面 - 可扩展】
		HSSFCellStyle columnTopStyle = this.getColumnTopStyle(workbook);// 获取列头样式对象
		HSSFCellStyle style = this.getStyle(workbook); // 单元格样式对象

		HSSFRow headRow = null;
		// 产生表格标题行
		if (title != null) {
			HSSFRow titleRow = sheet.createRow(0);
			HSSFCell titleCell = titleRow.createCell(0);
			sheet.addMergedRegion(new CellRangeAddress(0, 1, 0, (rowName.length - 1))); // 设置标题跨行和跨列
			titleCell.setCellStyle(columnTopStyle);
			titleCell.setCellValue(title);
			headRow = sheet.createRow(2); // 有标题,在索引2的位置创建行(最顶端的行开始的第二行)
		} else {
			headRow = sheet.createRow(0); // 没有标题,直接在第一行创建列头
		}

		// 定义所需列数
		int columnNum = rowName.length;

		// 将列头设置到sheet的单元格中
		for (int i = 0; i < columnNum; i++) {
			HSSFCell headCell = headRow.createCell(i); // 创建列头对应个数的单元格
			headCell.setCellType(HSSFCell.CELL_TYPE_STRING); // 设置列头单元格的数据类型
			RichTextString text = new HSSFRichTextString(rowName[i]);
			headCell.setCellValue(text); // 设置列头单元格的值
			headCell.setCellStyle(columnTopStyle); // 设置列头单元格样式
		}

		// 将查询出的数据设置到sheet对应的单元格中
		for (int i = 0; i < dataList.size(); i++) {
			int rowIndex = title == null ? 1 : 3;
			Object[] obj = dataList.get(i);// 遍历每个对象
			HSSFRow row = sheet.createRow(i + rowIndex);// 创建所需的行

			for (int j = 0; j < obj.length; j++) {
				HSSFCell cell = row.createCell(j, HSSFCell.CELL_TYPE_STRING); // 设置单元格的数据类型
				if (obj[j] == null) {
					cell.setCellValue("");
				} else {
					cell.setCellValue(obj[j].toString()); // 设置单元格的值
				}
				cell.setCellStyle(style); // 设置单元格样式
			}
		}
		// 让列宽随着导出的数据自动适应
		for (int colNum = 0; colNum < columnNum; colNum++) {
			int columnWidth = sheet.getColumnWidth(colNum) / 256;
			for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++) {
				HSSFRow currentRow = sheet.getRow(rowNum);
				if (currentRow != null) {
					HSSFCell currentCell = currentRow.getCell(colNum);
					if (currentCell != null) {
						if (currentCell.getCellType() == HSSFCell.CELL_TYPE_STRING) {
							int length = currentCell.getStringCellValue().length();
							if (columnWidth < length) {
								columnWidth = length;
							}
						}
					}
				}
			}
			sheet.setColumnWidth(colNum, (columnWidth + 6) * 256);
		}
		workbook.write(out);
	}

	/*
	 * 列头单元格样式
	 */
	public HSSFCellStyle getColumnTopStyle(HSSFWorkbook workbook) {

		// 设置字体
		HSSFFont font = workbook.createFont();
		// 设置字体大小
		font.setFontHeightInPoints((short) 11);
		// 字体加粗
		font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		// 设置字体名字
		font.setFontName("Courier New");
		// 设置样式;
		HSSFCellStyle style = workbook.createCellStyle();
		// 设置底边框;
		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		// 设置底边框颜色;
		style.setBottomBorderColor(HSSFColor.BLACK.index);
		// 设置左边框;
		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		// 设置左边框颜色;
		style.setLeftBorderColor(HSSFColor.BLACK.index);
		// 设置右边框;
		style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		// 设置右边框颜色;
		style.setRightBorderColor(HSSFColor.BLACK.index);
		// 设置顶边框;
		style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		// 设置顶边框颜色;
		style.setTopBorderColor(HSSFColor.BLACK.index);
		// 在样式用应用设置的字体;
		style.setFont(font);
		// 设置自动换行;
		style.setWrapText(false);
		// 设置水平对齐的样式为居中对齐;
		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		// 设置垂直对齐的样式为居中对齐;
		style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);

		return style;

	}

	/*
	 * 列数据信息单元格样式
	 */
	public HSSFCellStyle getStyle(HSSFWorkbook workbook) {
		// 设置字体
		HSSFFont font = workbook.createFont();
		// 设置字体大小
		// font.setFontHeightInPoints((short)10);
		// 字体加粗
		// font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		// 设置字体名字
		font.setFontName("Courier New");
		// 设置样式;
		HSSFCellStyle style = workbook.createCellStyle();
		// 设置底边框;
		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		// 设置底边框颜色;
		style.setBottomBorderColor(HSSFColor.BLACK.index);
		// 设置左边框;
		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		// 设置左边框颜色;
		style.setLeftBorderColor(HSSFColor.BLACK.index);
		// 设置右边框;
		style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		// 设置右边框颜色;
		style.setRightBorderColor(HSSFColor.BLACK.index);
		// 设置顶边框;
		style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		// 设置顶边框颜色;
		style.setTopBorderColor(HSSFColor.BLACK.index);
		// 在样式用应用设置的字体;
		style.setFont(font);
		// 设置自动换行;
		style.setWrapText(false);
		// 设置水平对齐的样式为居中对齐;
		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		// 设置垂直对齐的样式为居中对齐;
		style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);

		return style;

	}
}
